from torch.utils.data import Dataset


class StockDataSet(Dataset):

    def __init__(self, seq_data, value_data):
        self.seq_data = seq_data
        self.value_data = value_data

    def __len__(self):
        return len(self.seq_data)

    def __getitem__(self, idx):
        return self.seq_data[idx], self.value_data[idx]
